package com.aiinterview.subjectservice.service.impl;


import com.aiinterview.common.entity.PageResult;
import com.aiinterview.model.subject.entity.SubjectCategory;
import com.aiinterview.subjectservice.mapper.SubjectCategoryDao;
import com.aiinterview.subjectservice.service.SubjectCategoryService;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 题目分类(SubjectCategory)表服务实现类
 *
 * @author makejava
 * @since 2024-12-25 11:15:41
 */
@Service("subjectCategoryService")

@Slf4j
public class SubjectCategoryServiceImpl implements SubjectCategoryService {
    @Resource
    private SubjectCategoryDao subjectCategoryDao;

    /**
     * 通过ID查询单条数据
     *
     * @param id 主键
     * @return 实例对象
     */
    @Override
    public SubjectCategory queryById(Long id) {
        return this.subjectCategoryDao.queryById(id);
    }


    /**
     * 新增数据
     *
     * @param subjectCategory 实例对象
     * @return 实例对象
     */
    @Override
    public SubjectCategory insert(SubjectCategory subjectCategory) {
        this.subjectCategoryDao.insert(subjectCategory);
        return subjectCategory;
    }

    /**
     * 修改数据
     *
     * @param subjectCategory 实例对象
     * @return 实例对象
     */
    @Override
    public SubjectCategory update(SubjectCategory subjectCategory) {
        this.subjectCategoryDao.update(subjectCategory);
        return this.queryById(subjectCategory.getId());
    }

    /**
     * 通过主键删除数据
     *
     * @param id 主键
     * @return 是否成功
     */
    @Override
    public boolean deleteById(Long id) {
        return this.subjectCategoryDao.deleteById(id) > 0;
    }

    /**
     * 获取题目顶级分类
     *
     * @return
     */
    @Override
    public List<SubjectCategory> getTopCategory() {
        return subjectCategoryDao.getTopCategory();
    }


    /**
     * 根据父ID查询出所有有关的子分类
     *
     * @param parentId 父ID
     * @return
     */
    @Override
    public List<SubjectCategory> getCategoryListByParentId(String parentId) {
        // 查询所有分类数据
        List<SubjectCategory> allCategories = subjectCategoryDao.getAllCategories();
        if (allCategories == null || allCategories.isEmpty()) {
            return new ArrayList<>(); // 如果查询结果为 null 或为空，返回空列表
        }

        // 建立 parentId 到子分类的索引
        Map<String, List<SubjectCategory>> parentIdToCategoriesMap = allCategories.stream()
                .filter(Objects::nonNull) // 过滤掉 null 对象
                .filter(category -> category.getParentId() != null) // 过滤掉 parentId 为 null 的分类
                .collect(Collectors.groupingBy(
                        category -> String.valueOf(category.getParentId()) // 按 parentId 分组
                ));

        // 结果列表
        List<SubjectCategory> resultList = new ArrayList<>();

        // 使用队列进行广度优先搜索（BFS）
        Queue<String> queue = new ArrayDeque<>();
        if (parentId != null) {
            queue.add(parentId);
        } else {
            log.warn("传入的 parentId 为 null，返回空列表");
            return resultList;
        }

        while (!queue.isEmpty()) {
            String currentId = queue.poll();
            if (currentId == null) {
                log.warn("当前 currentId 为 null，跳过处理");
                continue; // 忽略 null 的 currentId
            }

            // 从索引中获取当前分类的子分类
            List<SubjectCategory> childCategories = parentIdToCategoriesMap.getOrDefault(currentId, Collections.emptyList());
            // 将子分类加入结果列表
            resultList.addAll(childCategories);
            // 将子分类的ID加入队列，继续查询下一级子分类
            childCategories.stream()
                    .filter(Objects::nonNull) // 过滤掉 null 对象
                    .map(SubjectCategory::getId) // 获取 ID
                    .filter(Objects::nonNull) // 过滤掉 null ID
                    .map(String::valueOf) // 转为字符串
                    .forEach(queue::add);

        }

        // 记录日志
        log.info("Service层查询父分类ID为 {} 的所有子分类，共 {} 条结果", parentId, resultList.size());
        return resultList;
    }

    @Override
    public PageResult pageQuery(int pageIndex, int pageSize) {
        PageHelper.startPage(1, 1);
        Page<SubjectCategory> page = subjectCategoryDao.pageCategory();
        return new PageResult(page.getTotal(), page.getResult());
    }
}
